import re
import time

from selenium import webdriver
from selenium.webdriver.common.by import By


def startUp():
    # 驱动浏览器（第一个字母必须大写）
    driver = webdriver.Chrome()
    # 访问目标网址（必须是以http开头的全网址）
    driver.get("http://www.baidu.com")
    # 最大化浏览器页面
    driver.maximize_window()
    # time.sleep(3)
    # # 后退（访问上一步访问的网址）
    # driver .back()
    # time.sleep(3)
    # # 前进（访问后一步访问的网址）
    # driver.forward()
    # time.sleep(3)
    # # 刷新浏览器
    # driver.refresh()
    # time.sleep(3)
    # # 关闭当前页面
    # driver.close()
    # 截图
    # driver.get_screenshot_as_file("./baidu.png")
    # png = driver.get_screenshot_as_png()
    # with open("./baidu2.png","wb")as fp:
    #     fp.write(png)
    # 获取元素的指定属性
    res = driver.find_element(By.ID, "hotsearch-content-wrapper").get_attribute("innerHTML")
    print(res)
    s1 = re.findall('class="title-content-title">(.*?)</span>', res)
    print(s1)
    time.sleep(3)
    # 退出浏览器
    driver.quit()


if __name__ == '__main__':
    startUp()
